typeerror 'float' object cannot be interpreted as an integer np.linspace